# -*- coding: utf-8 -*-
# @Author  : JackRoc
# @FileName: iframe.py

from selenium import webdriver
import time
import os
driver = webdriver.Chrome()
file_path = 'file:///' + os.path.abspath('frame.html')
driver.get(file_path)
#****切换到iframe（name = "iframename"）
##switch_to_frame()方法不赞成使用，使用switch_to.frame()方法代替
##存在iframe页面嵌套，但是没有切换，会出现无法定位到指定元素
driver.switch_to.frame("iframename")
#****下面就可以正常的操作元素了
driver.find_element_by_id("firstname").send_keys("firstname")
time.sleep(3)
driver.quit()


#页面截图：用于捕捉当前浏览上下文的截图，截图返回以Base64格式编码的截图。
#可以使用driver.save_screenshot(),
from selenium import webdriver
driver = webdriver.Chrome()
# Navigate to url
driver.get("https://www.baidu.com")
# Returns and base64 encoded string into image
driver.save_screenshot('./baidu.png')
driver.quit()

#元素截图：
#用于捕获当前浏览上下文的元素截图。截图返回以Base64格式编码的截图。

from selenium import webdriver
import time
driver = webdriver.Chrome()
driver.get('https://www.baidu.com')
ele = driver.find_element_by_id('kw')
ele.screenshot('./baidu2.png')
##driver.save_screenshot('./baidu.png')
time.sleep(2)
driver.quit()

"""
hello 不好意思打扰你
很荣幸你能观看到此视频
给大家讲一个故事 
"""